import pymysql


class DB:
    def __init__(self):
        self.conn = pymysql.connect(host='localhost', port=3306, user='root', password='root', charset='utf8',
                                    db='workf', cursorclass=pymysql.cursors.DictCursor)
        self.cursor = self.conn.cursor()

    # 添加，修改，删除
    def insert(self, sql):
        self.cursor.execute(sql)
        self.conn.commit()

    def find(self, sql):
        self.cursor.execute(sql)
        res= self.cursor.fetchone()
        return res


db1 = DB()
